tensor([[[0.4200, 0.4200],
         [0.4200, 0.4200]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]]], grad_fn=<RoundBackward1>)